Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Improve reduction schedule on arm CPUs #6110

Merged
merged 2 commits into from
Jul 23, 2020

Conversation

giuseros
Copy link
Contributor

This improves reduction by ~10x using x86 schedules. While for float32 reduction is not common, it is widely used in quantized operations (int8, uint8) in order to compute the offset contributions.

We might revisit this later to tailor the schedule for Arm CPUs.

Change-Id: I9cd85deac6a57666b82ff7250d827652a4000d82

Change-Id: I9cd85deac6a57666b82ff7250d827652a4000d82
@giuseros
Copy link
Contributor Author

cc @u99127 @anijain2305

@anijain2305
Copy link
Contributor

Can you please retrigger the CI. It is failing due to a known issue - #6084

Change-Id: I5efd99e34268e6bb990904a4b98e1edf2174b26b
@giuseros
Copy link
Contributor Author

Hi @anijain2305 ,

Thanks for the approval! I did as you said and re-triggered the CI

@anijain2305 anijain2305 merged commit 9d34eaa into apache:master Jul 23, 2020
@anijain2305
Copy link
Contributor

Thanks @giuseros This is merged!

trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Aug 26, 2020
* Improve reduction schedule on arm CPUs

Change-Id: I9cd85deac6a57666b82ff7250d827652a4000d82

* Retrigger CI

Change-Id: I5efd99e34268e6bb990904a4b98e1edf2174b26b
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Aug 26, 2020
* Improve reduction schedule on arm CPUs

Change-Id: I9cd85deac6a57666b82ff7250d827652a4000d82

* Retrigger CI

Change-Id: I5efd99e34268e6bb990904a4b98e1edf2174b26b
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Sep 2, 2020
* Improve reduction schedule on arm CPUs

Change-Id: I9cd85deac6a57666b82ff7250d827652a4000d82

* Retrigger CI

Change-Id: I5efd99e34268e6bb990904a4b98e1edf2174b26b
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Sep 3, 2020
* Improve reduction schedule on arm CPUs

Change-Id: I9cd85deac6a57666b82ff7250d827652a4000d82

* Retrigger CI

Change-Id: I5efd99e34268e6bb990904a4b98e1edf2174b26b
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants